package com.lmf.rabbitmq.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * 高级确认发布：交换机回调方法的实现，是RabbitTemplate的内部接口，确认交换机发布接口，回退消息接口
 *
 * 注意：备份交换机比消息回退机制优先级高
 */
@Slf4j
@Component
public class MyCallBack implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnsCallback{

    @Autowired
    private RabbitTemplate rabbitTemplate;

    @PostConstruct  //在类加载初始化时，调用构造方法之后运行被@PostConstruct注解修饰的方法
    public void init(){
        //将确认回调接口的实现类，注入rabbitTemplate中去
        rabbitTemplate.setConfirmCallback(this);
    }

    @PostConstruct
    public void init2(){
        rabbitTemplate.setReturnsCallback(this);
    }

    /**
     * 交换机不管是否收到消息,都会调用的一个回调方法
     *
     * @param correlationData 消息相关数据
     * @param ack             交换机是否收到消息
     * @param cause           为收到消息的原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String id = correlationData != null ? correlationData.getId() : "";
        if (ack) {
            log.info("交换机：已经收到 id 为:{}的消息", id);
        } else {
            log.info("交换机：还未收到 id 为:{}消息，原因:{}", id, cause);
        }
    }

    /**
     * 当消息无法路由的时候的回调方法：即，交换机找不到合适的队列投递消息的时候，就会将该消息回退给生产者，让生产者自己处理
     * @param returned
     */
    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.error("交换机：消息：{}，被交换机 {} 退回，原因：{}，路由key：{},code:{}",
                new String(returned.getMessage().getBody()),
                returned.getExchange(),
                returned.getReplyText(),
                returned.getRoutingKey(),
                returned.getReplyCode());
    }

}
